package bintree.leetcode;

import java.util.Deque;
import java.util.LinkedList;

/**
 * 判断完全二叉树
 * @author yuisama
 * @date 2022/05/13 19:59
 **/
public class Num958_IsCompleteTree {
    public boolean isCompleteTree(TreeNode root) {
        Deque<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        // 判断是否是第二阶段
        boolean isSecondStep = false;
        while (!queue.isEmpty()) {
            TreeNode cur = queue.poll();
            if (!isSecondStep) {
                // 在第一阶段中，每个节点都必须存在左右子树
                if (cur.left != null && cur.right != null) {
                    queue.offer(cur.left);
                    queue.offer(cur.right);
                }else if (cur.left != null) {
                    // 此时只有左树没有右树，转阶段
                    isSecondStep = true;
                    queue.offer(cur.left);
                }else if (cur.right != null) {
                    // 只有右树没左树，反例
                    return false;
                }else {
                    // 碰到叶子结点，转阶段
                    isSecondStep = true;
                }
            }else {
                // 此时处在第二阶段，所有节点都必须为叶子结点。
                if (cur.left != null || cur.right != null) {
                    // 不是叶子结点，反例
                    return false;
                }
            }
        }
        return true;
    }
}